#!/usr/bin/env python
# -*- coding:utf-8 -*-
# @FileName  :1.5 udp聊天器优化.py
# @Time      :2020/7/9 16:23
# @Author    :亮亮
# @说明       :
# @总结       :
import socket

# 输入发送方的IP和端口
dest_ip = input('输入对方IP:')
dest_port = int(input('输入对方端口:'))


def send_msg(udp_data):
    """发送数据"""

    # 输入发送的内容
    sent_data = input('输入要发送的内容:')

    # 发送
    udp_data.sendto(sent_data.encode('utf-8'), (dest_ip, dest_port))


def recv_msg(udp_data):
    """接收数据"""
    # 接收显示
    recv_data = udp_data.recvfrom(1024)

    return recv_data[0].decode('utf-8')


def main():
    """主函数"""
    # 创建socket套接字
    udp_data = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

    # 绑定端口和IP
    local_addr = ('', 9091)
    udp_data.bind(local_addr)

    # 创建循环
    while True:
        # 发送数据
        send_msg(udp_data)

        # 接收数据
        print(recv_msg(udp_data))


if __name__ == '__main__':
    main()
